import tensorflow as tf


if __name__ == "__main__":
    x = tf.zeros([3, 3], dtype=tf.int32)
    print(x)
    # Crop 0 axis.
    y = tf.split(x, axis=0, num_or_size_splits=[1, 2])
    print(y[1].shape)
    # Crop 0,1 axis, store left corner.
    y1 = tf.split(x, axis=0, num_or_size_splits=[1, 2])
    y2 = tf.split(y1[0], axis=1, num_or_size_splits=[2, 1])
    print(len(y2[0].numpy().shape))

